package com.example.leetcode.trainingcamp.week9.sunday;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * 你这个学期必须选修 numCourses 门课程，记为 0 到 numCourses - 1 。
 *
 * 在选修某些课程之前需要一些先修课程。 先修课程按数组 prerequisites 给出，其中 prerequisites[i] = [ai, bi] ，表示如果要学习课程 ai 则 必须 先学习课程  bi 。
 *
 * 例如，先修课程对 [0, 1] 表示：想要学习课程 0 ，你需要先完成课程 1 。
 * 请你判断是否可能完成所有课程的学习？如果可以，返回 true ；否则，返回 false 。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/course-schedule
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class Test207 {

    private final List<List<Integer>> list = new ArrayList<>();
  // 拓扑排序，统计入度为0的点
    public boolean canFinish(int numCourses, int[][] prerequisites) {
        for (int i =0;i<numCourses;i++){
            list.add(new ArrayList<>());
        }
        int[] visit = new int[numCourses];
        for (int[]p:prerequisites) {
            list.get(p[1]).add(p[0]);
            ++visit[p[0]];
        }
        Queue<Integer> queue = new LinkedList<>();
        for (int i =0;i<numCourses;i++){
            if (visit[i] ==0 ) queue.add(i);
        }
        int count = 0;
        while (!queue.isEmpty()){
            Integer poll = queue.poll();
            count++;
            for (int i:list.get(poll)) {
                --visit[i];
                if (visit[i] == 0) queue.offer(i);
            }
        }
        return count==numCourses;
    }
}
